package com.example.wudi.config;//package com.example.wudi.config;
//
//import cn.dev33.satoken.exception.NotLoginException;
//import cn.dev33.satoken.exception.NotRoleException;
//import com.example.wudi.res.Res;
//import org.springframework.web.bind.annotation.ExceptionHandler;
//import org.springframework.web.bind.annotation.RestControllerAdvice;
//
//@RestControllerAdvice
//public class SaTokenExceptionHandler {
//
//    @ExceptionHandler(NotLoginException.class)
//    public Res handlerNotLoginException(NotLoginException nle) {
//        System.out.println("asdakjsdbkabdkja" + nle.getType());
//        // 不同异常返回不同状态码
//        String message = "";
//        if (nle.getType().equals(NotLoginException.NOT_TOKEN)) {
//            message = "未提供Token";
//        } else if (nle.getType().equals(NotLoginException.INVALID_TOKEN)) {
//            message = "未提供有效的Token";
//        } else if (nle.getType().equals(NotLoginException.TOKEN_TIMEOUT)) {
//            message = "登录信息已过期，请重新登录";
//        } else if (nle.getType().equals(NotLoginException.BE_REPLACED)) {
//            message = "您的账户已在另一台设备上登录，如非本人操作，请立即修改密码";
//        } else if (nle.getType().equals(NotLoginException.KICK_OUT)) {
//            message = "已被系统强制下线";
//        } else {
//            message = "当前会话未登录";
//        }
//        // 返回给前端
//        return Res.noBig(message, null );
//    }
//
//}
//
//

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

import com.example.wudi.res.Res;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.ResponseBody;

import cn.dev33.satoken.exception.DisableLoginException;
import cn.dev33.satoken.exception.NotLoginException;
import cn.dev33.satoken.exception.NotPermissionException;
import cn.dev33.satoken.exception.NotRoleException;

/**
 * 全局异常处理
 */
@ControllerAdvice
public class SatokenExcptionHandler {

    // 全局异常拦截（拦截项目中的所有异常）
    @ResponseBody
    @ExceptionHandler
    public Res handlerException(Exception e, HttpServletRequest request, HttpServletResponse response)
            throws Exception {

        // 打印堆栈，以供调试
        System.out.println("全局异常---------------");
        e.printStackTrace();

        // 不同异常返回不同状态码
        String aj = null;
        if (e instanceof NotLoginException) {	// 如果是未登录异常
            NotLoginException ee = (NotLoginException) e;
            aj = ee.getMessage();
        }
        else if(e instanceof NotRoleException) {		// 如果是角色异常
            NotRoleException ee = (NotRoleException) e;
            aj = "无此角色：" + ee.getRole();
        }
        else if(e instanceof NotPermissionException) {	// 如果是权限异常
            NotPermissionException ee = (NotPermissionException) e;
            aj = "无此权限：" + ee.getCode();
        }
        else if(e instanceof DisableLoginException) {	// 如果是被封禁异常
            DisableLoginException ee = (DisableLoginException) e;
            aj = "账号被封禁：" + ee.getDisableTime() + "秒后解封";
        }
        else {	// 普通异常, 输出：500 + 异常信息
            aj = e.getMessage();
        }

        // 返回给前端
        return Res.noBig(aj, null);
    }

}
